import cv2
import numpy as np
import torch

val0 = torch.tensor([
    [1, 2, 3],
    [4, 5, 6],
]).view(1, 1, 2, 3)
print(val0)
